import pymysql

def main():
    no = int(input('要更新的部门编号:'))
    loc = input('部门新地址:')
    conn = pymysql.connect(host='192.168.84.89', port=3306,
                           user='root', password='Zhjai1@131waaA23',
                           db='hrs', charset='utf8')
    try:
        with conn.cursor() as cursor:
            result=cursor.execute(
                'update tb_dept set dloc=%s where dno=%s',(loc,no))
            if result==1:
                print('更新成功了')
                conn.commit()
    except pymysql.MySQLError as error:
        print(error)
        conn.rollback()
    finally:
        conn.close()
if __name__ == '__main__':
    main()
